package 剑指Offer1;

import java.util.HashMap;

public class _05复杂链表的复制 {
    class Node {
        int val;
        Node next;
        Node random;

        public Node(int val) {
            this.val = val;
            this.next = null;
            this.random = null;
        }
    }

    public Node copyRandomList(Node head) {
        if(head == null ) return null;

        HashMap<Node,Node> map = new HashMap<>();
        Node tmp = head;
        //复制各节点，并建立 “原节点 -> 新节点” 的 Map 映射
        while(tmp!=null){
            map.put(tmp,new Node(tmp.val));
            tmp = tmp.next;
        }
        tmp = head;
        //构建新链表的 next 和 random 指向
        while(tmp!=null){
            //将存在哈希表的节点串联起来
           map.get(tmp).next = map.get(tmp.next);
           map.get(tmp).random = map.get(tmp.random);
           tmp = tmp.next;
        }
        return map.get(head); //返回在哈希表中的头节点
    }
}
